#!/usr/bin/env python
# -*- coding: utf-8 -*-
# @Date    : 2025-01-31-18:37
# @Author  :
# @File    : train.py
# @Software: PyCharm


import warnings
# 忽略所有警告信息
warnings.filterwarnings('ignore')

from ultralytics import YOLO

"""
训练模型
"""

if __name__ == '__main__':
    """
     1、从YAML文件构建新模型
     model = YOLO("yolo11n.yaml")
     2、从预训练权重构建模型
     model = YOLO("yolo11n.pt")
     3、从YAML文件构建新模型，并将预训练权重转移到新模型
     model = YOLO("yolo11n.yaml").load("yolo11n.pt")
     
    """

    #  调用官方模型
    model = YOLO("yolo11n.pt")
    #     修改为自己的数据集地址
    results = model.train(
        data='ultralytics/datasets/original-license-plates.yaml',
        epochs=100, # 训练次数
        imgsz=640, # 输入的图片的大小
        device=0, # GPU 数
        project='runs/train',  # 输出路径
    )

